package com.fzn.jedis;

import redis.clients.jedis.Jedis;

import java.util.Random;

/**
 * @program: redis
 * 描述：  手机验证码
 * @author: fzn
 * @create: 2022-03-24 15:24
 **/
public class PhoneCode {
    public static void main(String[] args) {
        // 模拟验证码发送
        verifyCode("123456789");

        // 校验
        // getRedisCode("123456789","309976");

    }
    // 生成6位数字验证码
    public static String getCode() {
        Random random = new Random();
        StringBuilder code = new StringBuilder();
        for (int i = 0; i < 6; i++) {
            int rand = random.nextInt(10);
            code.append(rand);
        }
        return code.toString();
    }
    // 每一个手机每天只能发送三次验证码 验证码放入redis中 设置过期时间
    public static void verifyCode(String phone) {
        // 连接redis
        Jedis jedis = new Jedis("172.16.234.134",6379);
        // 拼接key
        String countKey = "VerifyCode"+phone+":count";
        // 验证码key
        String codeKey = "VerifyCode"+phone+":code";

        // 每个手机每次只能发送三次
        String count = jedis.get(countKey);
        if (count == null) {
            // 没有发送次数 第一次发送
            // 设置发送次数为1
            jedis.setex(countKey,24*60*60,"1");
        }else if (Integer.parseInt(count) <= 2){
            // 发送次数加1
            jedis.incr(countKey);
        }else if (Integer.parseInt(count) > 2){
            // 发送三次
            System.out.println("今天次数已经发送超出三次");
            jedis.close();
            return;
        }
        String vcode = getCode();
        jedis.setex(codeKey,120,vcode);
        jedis.close();
    }

    // 验证码校验
    public static void getRedisCode(String phone,String code) {
        // 从redis中获取验证码
        Jedis jedis = new Jedis("172.16.234.134",6379);
        String codeKey = "VerifyCode"+phone+":code";
        String redisCode = jedis.get(codeKey);
        // 判断
        if (redisCode.equals(code)) {
            System.out.println("成功");
        }else {
            System.out.println("失败");
        }

    }

}